import torch
from timm.models.convnext import convnext_small
from timm.models.swin_transformer_v2 import swinv2_small_window16_256

if __name__ == '__main__':
    # model = swinv2_small_window16_256()
    model = convnext_small()
    output = "convnext_s.pth"
    torch.save(model, output)